Floyd's triangle code in Python